#include <iostream>
#include <string>
#include <vector>
#include <functional>
#include "InetAddr.hpp"
#include "Route.hpp"
#include "ThreadPool.hpp"
#include "UdpServer.hpp"

using namespace ThreadPoolModule;

using task_t = std::function<void()>;

int main(int argc, char* argv[]){
    if(argc != 2){
        std::cerr << "Usage: " << argv[0] << "server_port" << std::endl;
        exit(1);
    }

    uint16_t server_port = std::stoi(argv[1]);
    Enable_Console_Log_Strategy();

    Route r;
    
    auto tp = ThreadPool<task_t>::GetInstance();

    std::unique_ptr<UdpServer> usvr = std::make_unique<UdpServer>(server_port, 
        [&r, &tp](int sockfd, const std::string& message, InetAddr& client){
            task_t t = std::bind(&Route::MessageRoute, &r, sockfd, message, client);
            tp->Enqueue(t);
        }
    );

    usvr->Init();
    usvr->Start();
    return 0;
}
